# Super parameters
# lr = 0
epochs = 1000
device_ids = [1]

# Train:
batch_size = 16

# Val:
batch_size = 16
val_freq = 50

# Dataset
TRAIN_PATH = '/home/zzc/dataSet/ImageNet_1000number/train'
VAL_PATH = '/home/zzc/dataSet/ImageNet_1000number/val'
format_train = 'JPEG'
format_val = 'JPEG'

# Display and logging:

# Saving checkpoints:
MODEL_PATH = '/home/zzc/deeplearn_test/HiDDeN/model/'
checkpoint_on_error = True
SAVE_freq = 50
# Load:
suffix = 'model.pt'
train_next = False
trained_epoch = 0